import os
import shutil                                  # 导入shutil模块用于删除目录
import json

# 数据写入对象
class DataFile:
    def __init__(self,data_list=None):
        # 创建datajson文件夹，存放json数据
        project_path = os.path.dirname(os.path.abspath('.'))
        self.json_path = os.path.join(project_path, 'datajson')             # json文件路径
        os.makedirs(self.json_path, exist_ok=True)
        if data_list:
            self.data_list = data_list
            # 初始化函数
            self.save_json()

    # 写入文件
    def save_json(self):
        # 生成文件名字
        file_list = os.listdir(self.json_path)
        file_name = len(file_list) + 1

        # 配置保存文件路径
        final_path = os.path.join(self.json_path, str(file_name) + '.json')

        # 保存文件
        with open(final_path, 'w') as f:
            f.write(json.dumps(self.data_list,indent=4))

    # 删除datajson文件夹下所有文件
    def remove_data(self):
        shutil.rmtree(self.json_path)
        os.makedirs(self.json_path, exist_ok=True)





if __name__ == '__main__':
    dataFile = DataFile()
    dataFile.remove_data()
